from ultralytics import YOLO

if __name__ == '__main__':
    model = YOLO(r"ultralytics/cfg/models/12/yolo12.yaml")
    model.train(data='./dataset/data.yaml',
                # cache="ram",
                epochs=300,
                imgsz=640,
                workers=4,
                batch=32,
                # device=0,
                amp=False,
                project='runs/yolo12/train',
                name='exp',
                )
